# pip install zhipuai 请先在终端进行安装

import zhipuai

zhipuai.api_key = "41805a5412196acfdd9d63a905f9039a.Z0TWy2173NoP1ZkI"
response = zhipuai.model_api.sse_invoke(
    model="chatglm_lite",
    prompt= [{"role":"user","content":"帮我写一篇小红书笔记，有一个有趣的标题，向大家介绍肯德基新品墨辣莎沙系列，文字贴近小红书用户风格。该笔记正文主要内容为墨辣莎沙汁汁厚牛堡：由芝士和经典墨西哥风味莎沙酱、和牛肉饼、培根和酸黄瓜组成，甜辣适中；墨辣莎沙汁汁嫩牛堡：双层嫩牛肉饼，搭配经典墨西哥风味莎沙酱，又嫩又多汁。\n正文格式如下：\n标题：参考“咬一口受不了！！kfc新品牛堡太顶了！！“，”肯德基诚不欺我！！这个牛肉堡爆汁巨好吃！”的风格，进行优化\n我宣布墨辣莎沙系列真是我的近期zui爱！真的....\n墨辣莎沙汁汁厚牛堡\n巨厚的...\n墨辣莎沙汁汁嫩牛堡\n双层的....\n两种口味闭眼不踩雷..."}],
    temperature= 0.9,
    top_p= 0.7,
    incremental=True
)

for event in response.events():
    if event.event == "add":
        print(event.data, end="")
    elif event.event == "error" or event.event == "interrupted":
        print(event.data, end="")
    elif event.event == "finish":
        print(event.data)
        print(event.meta, end="")
    else:
        print(event.data, end="")